
import os
from .heart_detect.no1 import run_no1
from .heart_detect.no2 import run_no2
from .heart_detect.no3 import run_no3
import shutil


# file_names = os.listdir("image/")
# image_path = os.path.join("image/", file_names[0])

def GetClassfy(image_path):
    run_no1(image_path)
    run_no2()
    return run_no3()


def Delete():
    dir = "lib/heart_detect/picture"
    shutil.rmtree(dir)
    path = "lib/heart_detect/class/roi"
    dir2 = os.listdir("lib/heart_detect/class/roi")
    for f in dir2:
        file_path = os.path.join(path, f)
        os.remove(file_path)
